import sys
sys.path.append("/data/python_project/qiufengfeng/nlp_tools")
from nlp_tools.tasks.abs_task_model import ABCTaskModel


import sys
if 'win' in sys.platform:
    save_path = r'E:\model_output\ner'
else:
    #save_path = '/home/qiufengfeng/nlp/competition/天池/CCKS2021中文NLP地址要素解析/globalpoint'

    save_path = r'/home/fanfanfeng/project/gitee/nlp_tools/example/ner/tmp/ner'

    #
    #save_path = '/home/qiufengfeng/nlp/competition/天池/CCKS2021中文NLP地址要素解析/globalpoint_data_agument'
loaded_model = ABCTaskModel.load_model(save_path)



while True:
    text_input = input("请输入句子：")
    result = loaded_model.predict_entities([text_input])
    print(result)

